from info.modules.web_html import html_blu
from flask import current_app, make_response
from flask_wtf import csrf


# 定义访问静态文件的视图函数
@html_blu.route("/<re(r'.*'):html_file_name>")
def get_html(html_file_name):
    """提供html文件"""
    # 如果html_file_name为""， 表示访问的路径是/ ,请求的是主页
    if not html_file_name:
        html_file_name = "index.html"

    # 如果资源名不是favicon.ico
    if html_file_name != "favicon.ico":
        html_file_name = "html/" + html_file_name

    # flask提供的返回静态文件的方法
    response = make_response(current_app.send_static_file(html_file_name))

    # 设置cookie值  创建一个csrf_token值
    response.set_cookie("csrf_token", csrf.generate_csrf())

    return response